import {AtpAgent} from '@atproto/api'

import {type AnyProfileView} from '#/types/bsky/profile'

type PResp = Awaited<ReturnType<AtpAgent['getProfile']>>

// based on https://github.com/Janpot/escape-html-template-tag/blob/master/src/index.ts

const ENTITIES: {
  [key: string]: string
} = {
  '&': '&amp;',
  '<': '&lt;',
  '>': '&gt;',
  '"': '&quot;',
  "'": '&#39;',
  '/': '&#x2F;',
  '`': '&#x60;',
  '=': '&#x3D;',
}

const ENT_REGEX = new RegExp(Object.keys(ENTITIES).join('|'), 'g')

function escapehtml(unsafe: Sub): string {
  if (Array.isArray(unsafe)) {
    return unsafe.map(escapehtml).join('')
  }
  if (unsafe instanceof HtmlSafeString) {
    return unsafe.toString()
  }
  return String(unsafe).replace(ENT_REGEX, char => ENTITIES[char])
}

type Sub = HtmlSafeString | string | (HtmlSafeString | string)[]

export class HtmlSafeString {
  private _parts: readonly string[]
  private _subs: readonly Sub[]
  constructor(parts: readonly string[], subs: readonly Sub[]) {
    this._parts = parts
    this._subs = subs
  }

  toString(): string {
    let result = this._parts[0]
    for (let i = 1; i < this._parts.length; i++) {
      result += escapehtml(this._subs[i - 1]) + this._parts[i]
    }
    return result
  }
}

export function html(parts: TemplateStringsArray, ...subs: Sub[]) {
  return new HtmlSafeString(parts, subs)
}

export const renderHandleString = (profile: AnyProfileView) =>
  profile.displayName
    ? `${profile.displayName} (@${profile.handle})`
    : `@${profile.handle}`

class HeadHandler {
  profile: PResp
  url: string
  constructor(profile: PResp, url: string) {
    this.profile = profile
    this.url = url
  }
  async element(element) {
    const view = this.profile.data

    const description = view.description
      ? html`
          <meta name="description" content="${view.description}" />
          <meta property="og:description" content="${view.description}" />
        `
      : ''
    const img = view.banner
      ? html`
          <meta property="og:image" content="${view.banner}" />
          <meta name="twitter:card" content="summary_large_image" />
        `
      : view.avatar
        ? html`<meta name="twitter:card" content="summary" />`
        : ''
    element.append(
      html`
        <meta property="og:site_name" content="Witchsky" />
        <meta property="og:type" content="profile" />
        <meta property="profile:username" content="${view.handle}" />
        <meta property="og:url" content="${this.url}" />
        <meta property="og:title" content="${renderHandleString(view)}" />
        ${description} ${img}
        <meta name="twitter:label1" content="Account DID" />
        <meta name="twitter:value1" content="${view.did}" />
        <link
          rel="alternate"
          href="at://${view.did}/app.bsky.actor.profile/self" />
      `,
      {html: true},
    )
  }
}

class TitleHandler {
  profile: PResp
  constructor(profile: PResp) {
    this.profile = profile
  }
  async element(element) {
    element.setInnerContent(renderHandleString(this.profile.data))
  }
}

class NoscriptHandler {
  profile: PResp
  constructor(profile: PResp) {
    this.profile = profile
  }
  async element(element) {
    const view = this.profile.data

    element.append(
      html`
        <div id="bsky_profile_summary">
          <h3>Profile</h3>
          <p id="bsky_display_name">${view.displayName ?? ''}</p>
          <p id="bsky_handle">${view.handle}</p>
          <p id="bsky_did">${view.did}</p>
          <p id="bsky_profile_description">${view.description ?? ''}</p>
        </div>
      `,
      {html: true},
    )
  }
}

export async function onRequest(context) {
  const agent = new AtpAgent({service: 'https://public.api.bsky.app/'})
  const {request, env} = context
  const origin = new URL(request.url).origin

  const base = env.ASSETS.fetch(new URL('/', origin))
  try {
    const profile = await agent.getProfile({
      actor: context.params.handleOrDID,
    })
    return new HTMLRewriter()
      .on(`head`, new HeadHandler(profile, request.url))
      .on(`title`, new TitleHandler(profile))
      .on(`noscript`, new NoscriptHandler(profile))
      .transform(await base)
  } catch (e) {
    console.error(e)
    return await base
  }
}